#! /user/bin/python3
# coding-utf-8
# @Time = 2025/4/15 11:37
# @Author:Poolaris

from selenium.webdriver.common.by import By

from base.Homebase import HomeBase
from base.ObjectMap import ObjectMap


class HomePage(HomeBase, ObjectMap):
    def get_balance(self, driver):
        """获取首页账户余额"""
        banlance_xpath = self.user_balance()
        return self.element_get(driver,
                                By.XPATH, banlance_xpath).text
